package com.lwp.i18n.util;

import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.HanyuPinyinVCharType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;


/**
 * 将中文转换为拼音
 * @author zqz
 *
 */
public class ChineseToPinyin {

    /**
     * 获取全拼拼音,如果长度大于40，那么就取每个汉字开头拼音字母；
     * @param src
     * @return
     */
    public static String getAll(String src){
        String result= getPingYin(src).replaceAll("\\W", "");
        if(result.length()>40){
            result = getPinYinHeadChar(src).replaceAll("\\W", "");
        }
        return result;
    }

    public static String get(String src){
        String result = null;
        boolean isHeader = false;
        if(src.length()>7){
            result = getPinYinHeadChar(src);
            isHeader = true;
        }else{
            result = getPingYin(src);
        }
        result = result.replaceAll("\\W", "");
        if(isHeader && result.length()<6){
            result=getPingYin(src).replaceAll("\\W", "");
        }
        return result;
    }

    /**
     * 首字母转大写
     * @param s
     * @return
     */
    public static String toUpperCaseFirstOne(String s){
        if(Character.isUpperCase(s.charAt(0)))
            return s;
        else
            return (new StringBuilder()).append(Character.toUpperCase(s.charAt(0))).append(s.substring(1)).toString();
    }

    /**
     *  将汉字转换为全拼
     * @param src
     * @return
     */
    public static String getPingYin(String src) {
        char[] t1 = null;
        t1 = src.toCharArray();
        String[] t2 = new String[t1.length];
        HanyuPinyinOutputFormat t3 = new HanyuPinyinOutputFormat();
        t3.setCaseType(HanyuPinyinCaseType.LOWERCASE);
        t3.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
        t3.setVCharType(HanyuPinyinVCharType.WITH_V);
        String t4 = "";
        int t0 = t1.length;
        try {
            boolean first = false;
            for (int i = 0; i < t0; i++) {
                // 判断是否为汉字字符
                if (Character.toString(t1[i]).matches(
                        "[\\u4E00-\\u9FA5]+")) {
                    t2 = PinyinHelper.toHanyuPinyinStringArray(t1[i], t3);
                    t4 += first?toUpperCaseFirstOne(t2[0]):t2[0];
                    first = true;
                } else
                    t4 += Character.toString(t1[i]);
            }
            return t4;
        } catch (BadHanyuPinyinOutputFormatCombination e1) {
            e1.printStackTrace();
        }
        return t4;
    }

    /**
     *  返回中文的首字母
     * @param str
     * @return
     */
    public static String getPinYinHeadChar(String str) {

        String convert = "";
        for (int j = 0; j < str.length(); j++) {
            char word = str.charAt(j);
            String[] pinyinArray = PinyinHelper.toHanyuPinyinStringArray(word);
            if (pinyinArray != null) {
                convert += pinyinArray[0].charAt(0);
            } else {
                convert += word;
            }
        }
        return convert;
    }

    /**
     *  将字符串转移为ASCII码
     * @param cnStr
     * @return
     */
    public static String getCnASCII(String cnStr) {
        StringBuffer strBuf = new StringBuffer();
        byte[] bGBK = cnStr.getBytes();
        for (int i = 0; i < bGBK.length; i++) {
            strBuf.append(Integer.toHexString(bGBK[i] & 0xff));
        }
        return strBuf.toString();
    }

    /**
     * 测试
     * @param args
     */
    public static void main(String[] args) {
        System.out.println(getPingYin("綦江1_qq县").replaceAll("\\W", ""));
        System.out.println(getPinYinHeadChar("綦江县"));
        System.out.println(getCnASCII("綦江县"));
        System.out.println(getAll("綦江县"));
        System.out.println(getPingYin("綦江县"));
    }
}